#include "UdpServer.hpp"
#include "User.hpp"


// ./server_udp localip localport
int main(int argc, char* argv[])
{
    if (argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " localport" << std::endl;
        Die(USAGE_ERR);
    }
    uint16_t port = std::stoi(argv[1]);

    ENABLE_CONSOLE_LOG();

    // 用户管理模块
    std::unique_ptr<UserManager> um = std::make_unique<UserManager>();

    // 网络模块
    std::unique_ptr<UdpServer> svr_uptr = std::make_unique<UdpServer>(port);
    
    svr_uptr->RegisterService(
        [&um](InetAddr& id){ um->AddUser(id); },
        [&um](InetAddr& id){ um->DelUser(id); },
        [&um](int sockfd, const std::string& message){ um->Router(sockfd, message); }
    );

    svr_uptr->InitServer();
    svr_uptr->Start();
    return 0;
}